"""
学生视图层
"""


from db import models

# 学生注册接口
def student_register_interface(username, password):
    # 调用学生类下面的select方法，实例化得到对象
    student_obj = models.Student.select(username)

    if student_obj:
        return False, f'当前学生：{username} 已存在！'

    # 调用学生类下面的保存数据方法
    student_obj = models.Student(username, password)
    student_obj.save()
    return True, f'学生：{username} 注册成功！'

# 选择学校接口
def choice_school_interface(school_name, student_name):
    # 判断当前学生是否选择过学校
    student_obj = models.Student.select(student_name)

    if student_obj.school:
        return False, f'学生：[{student_name}] 已经选择过学校了！'

    # 调用添加学生接口
    student_obj.add_school(school_name)
    return True, f'学生：[{student_name}]  选择学校：[{school_name}] 成功！'